package com.heima.wemedia.service.impl;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.heima.aliyun.scan.GreenScan;
import com.heima.aliyun.scan.ScanResult;
import com.heima.feigns.AdminFeign;
import com.heima.model.common.constants.message.PublishArticleConstants;
import com.heima.model.common.constants.wemedia.WemediaConstants;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.common.exception.CustException;
import com.heima.model.wemedia.pojos.WmNews;
import com.heima.utils.common.SensitiveWordUtil;
import com.heima.wemedia.mapper.WmNewsMapper;
import com.heima.wemedia.service.WmNewsAutoScanService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

@Service
@Slf4j
public class WmNewsAutoScanServiceImpl implements WmNewsAutoScanService {
    @Autowired
    private WmNewsMapper wmNewsMapper;

    @Value("${file.oss.web-site}")
    String webSite;

    @Autowired
    private AdminFeign adminFeign;

    @Autowired
    private GreenScan greenScan;

    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * 自动审核方法
     *
     * @param id
     */
    @Override
    public void autoScanWmNews(Integer id) {
        // 1 判断文章id是否为空
        if (id == null) {
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID, "待审核的文章不存在");
        }
        // 2 根据id查询自媒体文章
        WmNews wmNews = wmNewsMapper.selectById(id);
        if (wmNews == null) {
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST, "此id对应的文章不存在");
        }
        // 3 判断文章状态 必须是 (待审核) (此步骤同样可以避免重复消费)
        if (!WemediaConstants.WM_NEWS_SUMMIT_STATUS.equals(wmNews.getStatus())) {
            // 不是待审核状态
            log.info("当前文章的状态是:{}", wmNews.getStatus());
            return;
        }
        // 4 抽取文章中的 所有文本内容 所有的图片内容
        Map<String, Object> contentAndImages = handleTextAndImages(wmNews);
        // 5 DFA 进行子管理敏感词审核  有敏感词不通过 改为2  通过继续下一步
        boolean havSensitive = handleSensitive((String) contentAndImages.get("content"), wmNews);
        if (!havSensitive) {
            // 返回false 表示审核不成功 结束处理
            log.info("文章内存在敏感词,审核未通过");
            return;
        }
        // 6 阿里云的文本审核 有违规词汇 改为2  不确定/aliyun未调用成功 改为3 通过继续下一步
        boolean scanContent = scanContent((String) contentAndImages.get("content"), wmNews);
        if (!scanContent) {
            // 返回false 表示审核不成功 结束处理
            log.info("文章内存在敏感词,审核未通过");
            return;
        }
        // 7 阿里云的图片审核 有违规图片 改为2 不确定/aliyun未调用成功 改为3 通过继续下一步
        List<String> picture = (List<String>) contentAndImages.get("images");
        boolean scanImage = scanImages(picture, wmNews);
        if (!scanImage) {
            // 返回false 表示审核不成功 结束处理
            log.info("图片审核未通过");
            return;
        }
        // 8 将文章状态改为8 自动审核通过
        updateWmNews(WmNews.Status.SUCCESS.getCode(), "审核通过", wmNews);
        // 9 根据文章的发布时间,发送延迟消息.用于定时发布文章(9 已发布)
        // 9.1 获取文章的发布时间
        long publishTime = wmNews.getPublishTime().getTime(); // 发布时间
        long currentTime = new Date().getTime(); // 当前时间
        long remainTime = publishTime - currentTime; // 距离发布时间的剩余时间
        // 9.2 发布文章
        rabbitTemplate.convertAndSend(PublishArticleConstants.DELAY_DIRECT_EXCHANGE,
                PublishArticleConstants.PUBLISH_ARTICLE_ROUTE_KEY,
                wmNews.getId(),
                message -> {                       // 延时消息 设置
                    message.getMessageProperties().setHeader("x-delay", remainTime <= 0 ? 0 : remainTime);
                    return message;
                });
        log.info("立即发布文章通知成功发送，文章id : {}", wmNews.getId());
    }

    /**
     * aliyun图片扫描
     *
     * @param images
     * @param wmNews
     * @return
     */
    private boolean scanImages(List<String> images, WmNews wmNews) {
        boolean flag = false;// 默认图片没有问题
        try {
            // 调用图片审核方法
            ScanResult scanResult = greenScan.imageUrlScan(images);
            String suggestion = scanResult.getSuggestion();
            switch (suggestion) {
                case "block":
                    // 失败 状态要改为2
                    updateWmNews(WmNews.Status.FAIL.getCode(), "图片中包含违规信息" + scanResult.getLabel(), wmNews);
                    break;
                case "review":
                    // 人工 状态改为3
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "图片中有不确定因素,需要进一步人工审核", wmNews);
                    break;
                case "pass":
                    // 审核通过
                    flag = true;
                    break;
                default:
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用异常,需要人工审核", wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            // 人工审核
            updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用失败，需要进一步人工审核:", wmNews);
        }
        return flag;
    }

    /**
     * aliyun扫描文本
     *
     * @param content
     * @param wmNews
     * @return
     */
    private boolean scanContent(String content, WmNews wmNews) {
        boolean flag = false;
        try {
            ScanResult scanResult = greenScan.greenTextScan(content);
            // 返回的suggestion 有三种模式 block: 违规 review 需要进一步人工审核 pass 通过
            String suggestion = scanResult.getSuggestion();
            // 根据suggestion来选择进行具体操作
            switch (suggestion) {
                case "block":
                    // 失败 状态要改为2
                    updateWmNews(WmNews.Status.FAIL.getCode(), "文章中包含敏感词汇" + scanResult.getLabel(), wmNews);
                    break;
                case "review":
                    // 人工 状态改为3
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "文章中有不确定因素,需要进一步人工审核" + scanResult.getLabel(), wmNews);
                    break;
                case "pass":
                    // 审核通过
                    flag = true;
                    break;
                default:
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用异常,需要人工审核" + scanResult.getLabel(), wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return flag;
    }

    /**
     * 扫描是否有敏感词 有则返回false 没有就返回 true
     *
     * @param content
     * @param wmNews
     * @return
     */
    private boolean handleSensitive(String content, WmNews wmNews) {
        boolean flag = true;// 默认没有敏感词
        // 1 远程调用查询敏感词
        ResponseResult<List<String>> sensitives = adminFeign.sensitives();
        // 2 判断远程调用是否成功
        if (!sensitives.checkCode()) {
            CustException.cust(AppHttpCodeEnum.REMOTE_SERVER_ERROR, "远程调用失败");
        }
        // 3 获取敏感词 将敏感词转为DFA数据
        List<String> data = sensitives.getData();
        SensitiveWordUtil.initMap(data);
        // 4 基于DFA扫描,内容中是否存在敏感词
        Map<String, Integer> matchWords = SensitiveWordUtil.matchWords(content);
        // 5 判断是否存在敏感词 有 则将文章状态修改为2
        if (!CollectionUtils.isEmpty(matchWords)) {
            short status = WmNews.Status.SUBMIT.getCode();
            String reason = "文章中包含以下敏感词汇" + matchWords.toString();
            updateWmNews(status, reason, wmNews);
            flag = false;
        }
        return flag;
    }

    /**
     *
     */
    private void updateWmNews(Short status, String reason, WmNews wmNews) {
        wmNews.setStatus(status);
        wmNews.setReason(reason);
        wmNewsMapper.updateById(wmNews);
    }

    /**
     * 抽取文章中的 所有文本内容 所有的图片内容
     *
     * @param wmNews 文章对象
     * @return map 中存放 "content":"文本内容 string类型" ,"images":"图片集合 list<string>类型 以,间隔"
     */
    private Map<String, Object> handleTextAndImages(WmNews wmNews) {
        // 准备Map集合
        Map<String, Object> result = new HashMap<>();
        // 1 判断内容不能为空 并转为 list<Map>
        String contentJson = wmNews.getContent();
        if (StringUtils.isBlank(contentJson)) {
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID, "该文章的内容为空");
        }
        List<Map> contentList = JSON.parseArray(contentJson, Map.class);
        // 2 抽取文章中的所有文本内容
        // 2.1 抽取content中所有文本,并拼接成一个字符串
        // 原来存储的是 [{type:text,value: 文本},{type,image,value :url}...]
        String content = contentList.stream().filter(s -> "text".equals(s.get("type")))
                .map(s1 -> s1.get("value").toString())
                .collect(Collectors.joining("_hmtt_"));//加一个拼接符 拼接符是自定义的
        // 2.2 将文本内容和标题拼接成字符串
        String title = wmNews.getTitle();
        content += title + "_hmtt_";
        // 2.3 将总的文本信息存入map集合
        result.put("content", content);
        // 3 抽取文中所有的图片列表
        // 3.1 抽取content 中所有的图片 得到图片列表
        List<String> imagesList = contentList.stream().filter(s -> "image".equals(s.get("type")))
                .map(s1 -> s1.get("value").toString())
                .collect(Collectors.toList());
        // 3.2 抽取封面中的所有图片 得到图片列表
        //  url1,url2,url3 封面图片,不带前缀 需要在每个url前面要加一个 website
        String coverImagesStr = wmNews.getImages();
        if (StringUtils.isNotBlank(coverImagesStr)) {
            // 先按照 , 分割
/*            String[] split = coverImagesStr.split(",");
            // 每个url前面要加一个 website
            for (int i = 0; i < split.length; i++) {
                split[i] = webSite + split[i];
            }
            List<String> coverImages = Arrays.stream(split).collect(Collectors.toList());*/
            //每个url前面要加一个 website
            List<String> coverImages = Arrays.stream(coverImagesStr.split(","))
                    .map(url -> webSite + url)
                    .collect(Collectors.toList());
            // 3.3 将封面图片列表和内容图片列表合并
            imagesList.addAll(coverImages);
        }
        // 3.4 对图片集合进行去重
        imagesList = imagesList.stream().distinct().collect(Collectors.toList());
        // 3.5 将总的图片集合存放到result中
        result.put("images", imagesList);
        // 4 返回结果
        return result;
    }
}